#!/bin/bash

RED='\033[0;31m'
GREEN='\033[0;32m'
NC='\033[0m' # No Color

# 检查参数
if [ $# -eq 0 ]; then
    echo -e "${RED}错误：请指定地区代码参数，例如：./run.sh cn${NC}"
    exit 1
fi

# 第一步：下载并转换
echo "开始下载并转换IP地址库..."
sudo bash auto_download_convert.sh $1
if [ $? -ne 0 ]; then
    echo -e "${RED}错误：下载转换失败${NC}"
    exit 1
fi

# 第二步：更新 ipset 和 iptables 规则
echo "开始更新ipset和iptables规则..."
sudo bash check_update_ipset.sh $1
if [ $? -ne 0 ]; then
    echo -e "${RED}错误：规则更新失败${NC}"
    exit 1
fi

echo -e "${GREEN}所有操作完成${NC}"
exit 0